package com.sky.handler;

import com.sky.constant.MessageConstant;
import com.sky.exception.BaseException;
import com.sky.result.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.sql.SQLIntegrityConstraintViolationException;

/**
 * 全局异常处理器，处理项目中抛出的业务异常
 */
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {

    /**
     * 捕获业务异常
     *
     * @param ex
     * @return
     */
    @ExceptionHandler
    public Result<?> exceptionHandler(BaseException ex) {
        log.error("异常信息：{}", ex.getMessage());
        return Result.error(ex.getMessage());
    }

    /**
     * 处理SQL违反唯一约束异常
     *
     * @param ex
     * @return
     */
    @ExceptionHandler
    public Result<?> exceptionHandler(SQLIntegrityConstraintViolationException ex) {
        log.error("异常信息：{}", ex.getMessage());
        if (ex.getMessage().contains("Duplicate entry")) {
            String userName = ex.getMessage().split(" ")[2];
            return Result.error(userName + "已存在，请勿重复添加");
        }
        if (ex.getMessage().contains("null value in column")) {
            return Result.error("该字段不能为空，请检查后重新添加");
        }
        if (ex.getMessage().contains("Data too long for column")) {
            return Result.error("该字段长度超过限制，请检查后重新添加");
        }

        return Result.error(MessageConstant.UNKNOWN_ERROR);
    }

}
